from ultralytics import YOLO

import matplotlib

matplotlib.rc("font", family='Microsoft YaHei')

if __name__ == '__main__':
    model = YOLO('yolov10n.pt').cuda()
    model.train(data=r'./data/train.yaml', epochs=100, device=0, box=9)
    model.val()
